Removing the NEEDLE in the Haystack: Backdoor Removal in LLMs via Weight Orthogonalisation
从干草堆中拔针:通过权重正交化移除 LLM 后门
🧭 LLM 安全防御 · 权重正交化
Locai Labs · Centre for AI, Computer Science, UCL
训练免清洗方法估计后门方向与拒绝子空间,对权重正交化,在保持安全能力的同时实现最低 ASR
📌 工作参考价值(4 点)
- 若业务中微调或使用开源 LLM 作为导购/客服/推荐 Agent,上游权重可能被投毒;NEEDLE 提供一种无需干净基线和原始训练数据的应急修复手段,适合已上线模型的后门摘除。
- 其“估计激活方向 → 权重正交投影”的思路可推广到业务中的模型行为纠正:例如用少量对比样本估计生成式推荐模型中的偏见、幻觉或不合规输出方向,通过投影移除,无需全量微调。
- 保留拒绝子空间的做法可作为模型编辑的约束:在修改推荐/Agent 模型权重时,显式保持安全对齐相关表示不变,降低对拒绝、安全审核、对话质量等下游能力的附带损害。
- 评估体系可借鉴:线上模型任何补丁/编辑/校准,除了效果指标,应同时监控 ASR、KL 散度与能力基准,避免治好后门却牺牲推荐/对话质量。